#pragma once

#include <iostream>
#include <string>
#include <vector>
#include <atomic>
#include <fstream>
#include <sys/types.h>
#include <sys/stat.h>
#include <unistd.h>
#include <sys/time.h>
#include <boost/algorithm/string.hpp>

namespace ns_util
{
    class TimeUtil
    {
    public:
        // 获取当前时间，并转为字符串
        static std::string GetTimeStamp()
        {
            time_t curr_time = time(nullptr);

            // 将时间转为时间结构体
            struct tm *format_time = localtime(&curr_time);

            // 转换失败，可能为无效时间
            if (format_time == nullptr)
                return "None";

            char time_buffer[1024];

            // 转为想要的字符串格式放入缓冲区中
            snprintf(time_buffer, sizeof(time_buffer), "%d-%d-%d %d-%d-%d",
                     format_time->tm_year + 1900,
                     format_time->tm_mon + 1,
                     format_time->tm_mday,
                     format_time->tm_hour,
                     format_time->tm_min,
                     format_time->tm_sec);

            return time_buffer;
        }

        // 获取当前时间，以毫秒为单位
        static std::string GetTimeMs()
        {
            struct timeval _time;
            gettimeofday(&_time, nullptr);
            return std::to_string(_time.tv_sec * 1000 + _time.tv_usec / 1000);
        }
    };

    const std::string pathprefix = "./temp/"; // 路径前缀
    class PathUtil
    {
    public:
        // 构建完整路径 filename-> ./temp/filename.cpp
        static std::string AddSuffix(const std::string &filename, const std::string &suffix)
        {
            std::string path = pathprefix;
            path += filename;
            path += suffix;

            return path;
        }
        // 可执行程序
        static std::string Exe(const std::string &filename)
        {
            return AddSuffix(filename, ".exe");
        }

        // 标准错误流重定向（运行时错误信息）
        static std::string Stderr(const std::string &filename)
        {
            return AddSuffix(filename, ".stderr");
        }

        // 输入重定向
        static std::string Stdin(const std::string &filename)
        {
            return AddSuffix(filename, ".stdin");
        }

        // 输出重定向
        static std::string Stdout(const std::string &filename)
        {
            return AddSuffix(filename, ".stdout");
        }

        // 源文件
        static std::string Src(const std::string &filename)
        {
            return AddSuffix(filename, ".cpp");
        }

        // 编译错误信息
        static std::string CompileError(const std::string &filename)
        {
            return AddSuffix(filename, ".compile_error");
        }
    };

    class FileUtil
    {
    public:
        // 判断文件是否存在
        static bool IsFileExists(const std::string &pathname)
        {
            struct stat st;

            // 获取文件属性成功，则文件存在
            if (stat(pathname.c_str(), &st) == 0)
                return true;

            return false; // 获取失败，则文件不存在
        }

        // 生成唯一的文件名（仅文件名，不包含路径+后缀）
        static std::string UniqueFile()
        {
            static std::atomic_uint id(0);
            id++;
            std::string unique_id = std::to_string(id);
            std::string ms=TimeUtil::GetTimeMs();

            return ms+"_"+unique_id;
        }
        // 写入文件，target_file 是完整的路径名
        static bool WriteFile(const std::string &target_file, const std::string &content)
        {
            std::ofstream out(target_file);

            if (!out.is_open()) // 打开失败
                return false;

            // 开始写入
            out.write(content.c_str(), content.size());
            out.close();

            return true;
        }
        // 读取文件的内容
        // target_file：要读取的文件的路径，content：输出型参数，keep：是否保留\n
        static bool ReadFile(const std::string &target_file, std::string *content, bool keep = false)
        {
            (*content).clear();

            std::ifstream in(target_file);

            if (!in.is_open()) // 文件流打开失败
            {
                return false;
            }

            // 开始读取

            // getline 读取时不保存行分隔符，用 keep 来选择是否保留 \n
            std::string line;
            while (std::getline(in, line))
            {
                (*content) += line;
                (*content) += (keep ? "\n" : "");
            }

            in.close(); // 关闭流

            return true;
        }
    };

    class StringUtil
    {
    public:
        // 切割字符串 line，分割符为 sep，把切割后的结果放入 target 中，target 是输出型参数
        static void SplitString(const std::string &line,std::vector<std::string> *target,const std::string &sep)
        {
            boost::split((*target),line,boost::is_any_of(sep),boost::algorithm::token_compress_on);
        }
    };
}